package com.li.flink.datastream

import org.apache.flink.api.common.functions.Partitioner

class MyPartitioner extends Partitioner[Int] {
  /**
   * 自定义分区
   *
   * @param k 分区编号
   * @param i 分区总数
   * @return
   */
  override def partition(k: Int, i: Int): Int = {
    if (k % 2 == 0) {
      0
    } else {
      1
    }

  }
}
